package org.primefaces.examples.filter;

import java.io.IOException;
import javax.servlet.Filter;
import javax.servlet.FilterChain;
import javax.servlet.FilterConfig;
import javax.servlet.ServletException;
import javax.servlet.ServletRequest;
import javax.servlet.ServletResponse;
import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;

/**
 * Servlet Filter implementation class DownloadFileFilter
 */
public class DownloadFileFilter implements Filter {

    /**
     * Default constructor. 
     */
    public DownloadFileFilter() {
        // TODO Auto-generated constructor stub
    }

	/**
	 * @see Filter#destroy()
	 */
	public void destroy() {
		// TODO Auto-generated method stub
	}

	/**
	 * @see Filter#doFilter(ServletRequest, ServletResponse, FilterChain)
	 */
	public void doFilter(ServletRequest request, ServletResponse response, FilterChain chain) throws IOException, ServletException {
		// TODO Auto-generated method stub
		// place your code here

		response.setContentType("application/octet-stream");
		response.setContentType("application/force-download");
		((HttpServletResponse) response).setHeader("Content-Transfer-Encoding", "binary");
		
		HttpServletRequest httpRequ = (HttpServletRequest) request;
		String parameter = httpRequ.getParameter("fileName");
		if(parameter != null) {
			 
			System.out.println("Downloading file  "  + parameter);
			
		}
		
		String[] split = httpRequ.getRequestURI().split("/");
		String fileName = ""; 
		if(split.length > 0 )
			fileName =  split[split.length - 1];
		
		if(fileName.endsWith(".txt"))
			((HttpServletResponse) response).setHeader("Content-Disposition","attachment;filename=\"" + fileName);

		
		chain.doFilter(request, response);
	}

	/**
	 * @see Filter#init(FilterConfig)
	 */
	public void init(FilterConfig fConfig) throws ServletException {
		// TODO Auto-generated method stub
	}

}
